


USE_GPU="2" # specify a GPU



export CUDA_VISIBLE_DEVICES=${USE_GPU} 

python experiments/train_basic.py \
--exp_name results/train_folding_kitti_debug --lr 0.0002 80 0.5 \
--phase train  --dataset_name kitti_mulobj --train_split train_5x5 \
--augmentation True  --grid_dims 45 45 --pointnet_mlp_dims 3 64 64 64 128 1024 \
--pointnet_fc_dims 1024 512 512 --pointnet_mlp_dolastrelu False \
--folding1_dims 514 512 512 3 --folding2_dims 515 512 512 3


#  --n_epoch 640 --save_epoch_freq 10 --print_freq 1 --pc_write_freq 1000 \
# --batch_size 256 
# --xyz_chamfer_weight 0.01 --lr_policy step
# --augmentation_theta 0 --augmentation_rotation_axis 0 --augmentation_flip_axis 1
# --xyz_loss_type 0  --encoder pointnetvanilla --decoder foldingnetvanilla 